package cn.gintone.generator;

import java.io.IOException;
import java.io.InputStream;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

import org.mybatis.generator.api.MyBatisGenerator;
import org.mybatis.generator.config.Configuration;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.exception.InvalidConfigurationException;
import org.mybatis.generator.exception.XMLParserException;
import org.mybatis.generator.internal.DefaultShellCallback;

/**
 * 读取MGB配置生成代码
 * @author he
 *
 */
public class Generator {
	public static void main(String[] args) {
		//MBG执行过程中的警告信息
		List<String> warnings = new ArrayList<String>();
		//当生成的代码重复时，覆盖原代码
		boolean overwrite = true;
		//读取MBG配置文件
		InputStream is = Generator.class.getResourceAsStream("/generator.xml");
		ConfigurationParser cp = new ConfigurationParser(warnings);
		try {
			Configuration config = cp.parseConfiguration(is);
			is.close();
			
			DefaultShellCallback callback = new DefaultShellCallback(overwrite);
			//创建MBG
			MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config,callback,warnings);
			//执行生成代码
			myBatisGenerator.generate(null);
			for(String warning:warnings) {
				System.out.println(warning);
			}
		} catch (XMLParserException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} catch (InvalidConfigurationException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} catch (SQLException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} catch (InterruptedException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
}
